package com.example.oss.service.impl;

import com.aliyun.oss.OSS;
import com.aliyun.oss.common.utils.BinaryUtil;
import com.aliyun.oss.model.MatchMode;
import com.aliyun.oss.model.PolicyConditions;
import com.example.oss.service.OssService;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import java.nio.charset.StandardCharsets;
import java.text.SimpleDateFormat;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.Date;
import java.util.LinkedHashMap;
import java.util.Map;

@Slf4j
@Service
public class OssServiceImpl implements OssService {

    @Resource
    private OSS ossClient;

    @Value("${alibaba.cloud.access-key}")
    private String accessId;

    @Value("${alibaba.cloud.oss.endpoint}")
    private String endpoint;

    @Value("${alibaba.cloud.oss.bucket-name}")
    private String bucketName;

    @Value("${alibaba.cloud.oss.expire-time}")
    private Long expireTime;


    /*
    * 后端服务生成OSS上传签名，在前端使用签名直接上传文件到OSS。
    * 需要提前为 bucket 开启CORS跨域访问功能。开启地址为 OSS控制台 -> Bucket -> 数据安全 -> 跨域设置 -> 创建规则。
    */
    @Override
    public Map<String, String> generatePolicy() {
        log.info("开始生成OSS上传签名,生成时间: {}", LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss")));
        // 填写Host地址，格式为 https://bucketname.endpoint。
        String host = "https://" + bucketName + "." + endpoint;
        // 设置上传到OSS文件的前缀，可置空此项。置空后，文件将上传至Bucket的根目录下。
        String format = new SimpleDateFormat("yyyy-MM-dd").format(new Date());
        String dir = format + "/";
        // 设置上传文件的过期时间，单位秒。
        long expireEndTime = System.currentTimeMillis() + expireTime * 1000;

        Date expiration = new Date(expireEndTime);
        PolicyConditions policyConds = new PolicyConditions();
        policyConds.addConditionItem(PolicyConditions.COND_CONTENT_LENGTH_RANGE, 0, 1048576000);
        policyConds.addConditionItem(MatchMode.StartWith, PolicyConditions.COND_KEY, dir);

        String postPolicy = ossClient.generatePostPolicy(expiration, policyConds);
        byte[] binaryData = postPolicy.getBytes(StandardCharsets.UTF_8);
        String encodedPolicy = BinaryUtil.toBase64String(binaryData);
        String postSignature = ossClient.calculatePostSignature(postPolicy);

        Map<String, String> respMap = new LinkedHashMap<>();
        respMap.put("accessId", accessId);
        respMap.put("policy", encodedPolicy);
        respMap.put("signature", postSignature);
        respMap.put("dir", dir);
        respMap.put("host", host);
        respMap.put("expire", String.valueOf(expireEndTime / 1000));

        return respMap;
    }
}

// 服务端签名给前端，然后前端使用签名上传文件到 OSS 文件不经过服务端中转，前端使用签名直接上传到 OSS。
// 在客户端直接上传文件到OSS, 参考阿里云官方文档  https://help.aliyun.com/zh/oss/use-cases/java-1?spm=a2c4g.11186623.0.i37
// 前端相应的处理逻辑参考 https://github.com/aliyun/aliyun-oss-js-sdk/blob/master/samples/browser/upload.html